package com.shiyipicture.shiyipicturebackend.api.imageSearch.sub;

import com.shiyipicture.shiyipicturebackend.exception.BusinessException;
import com.shiyipicture.shiyipicturebackend.exception.ErrorCode;
import lombok.extern.slf4j.Slf4j;
import org.jsoup.Jsoup;
import org.jsoup.nodes.Document;
import org.jsoup.nodes.Element;
import org.jsoup.select.Elements;

import java.io.IOException;
import java.util.regex.Matcher;
import java.util.regex.Pattern;

@Slf4j
public class GetImageFirstUrlApi {

  /**
   * 获取图片列表页面地址
   *
   * @param url
   * @return
   */
  public static String getImageFirstUrl(String url){
    try {
      Document document = Jsoup.connect(url)
        .timeout(5000)
        .get();

      //获取所有的script标签
      Elements scriptElements = document.getElementsByTag("script");
      for (Element scriptElement : scriptElements) {
        String scriptContent = scriptElement.html();
        if (scriptContent.contains("\"firstUrl\"")) {
          // 正则表达式提取 firstUrl 的值
          Pattern pattern = Pattern.compile("\"firstUrl\"\\s*:\\s*\"(.*?)\"");
          Matcher matcher = pattern.matcher(scriptContent);
          if (matcher.find()) {
            String firstUrl = matcher.group(1);
            // 处理转义字符
            firstUrl = firstUrl.replace("\\/", "/");
            return firstUrl;
          }
        }
      }
      throw new BusinessException(ErrorCode.OPERATION_ERROR,"未找到url");
    } catch (IOException e) {
      log.error("搜索失败", e);
      throw new BusinessException(ErrorCode.OPERATION_ERROR, "搜索失败");
    }
  }

  @Deprecated
  public static void main(String[] args) {
    // 请求目标 URL
    String url = "https://graph.baidu.com/s?card_key=&entrance=GENERAL&extUiData[isLogoShow]=1&f=all&isLogoShow=1&session_id=13040450141592311466&sign=12182e97cd54acd88139901737014314&tpl_from=pc";
    String imageFirstUrl = getImageFirstUrl(url);
    System.out.println("搜索成功，结果 URL：" + imageFirstUrl);
  }
}
